#ATM机的存取款功能
#ATM角色 50000
#用户角色 5000
#1、先登录
#   根据用户登录
#2、菜单界面
#   1、取款
#      用户的账户金额就减少了500
#   2、存款
#      用户的账户金额就增加了500块
#   3、查询
#      用户的账户金额显示

# 注册函数
def register():
     user_name=input("请输入你要注册的用户名：\n")
     passwd=input("请输入你要设置的密码：\n")
     user_info[user_name]=passwd
     return login()

#登入函数
def login():
    user_name=input("用户名：\n")
    passwd=input("密码：\n")
    user_info["root"] = "1234"
    for i in user_info:
        if i==user_name and user_info[i]==passwd:
            print("登入成功！")
            return menu()
        else:print("用户名或者密码错误！")
#菜单函数
def menu():
    print("*"*20,"菜单界面","*"*20)
    global choice1
    choice1=input("请输入你的选择： # 1 取款  # 2 存款 # 3 查询 # 按任意健退出\n")
    if '1'<=choice1<='3':
        return menu_jud()
    else:exit()


def menu_jud():
    if choice1=='1':
       in_money=input("请输入你要取的金额：\n")
       if in_money.isdigit():
           in_money=int(in_money)
           if int(ATM_money['atm'])>=in_money: #ATM机余额判断
               money=int(user_money["root"])-in_money #剩余的余额
               ATM_money['atm']=int(ATM_money['atm'])-money  #取钱====atm机钱增加====从新赋值给ATm
               if money>=0:
                   print("操作成功！")
                   user_money['root']=money
                   return menu()
               elif money<0:
                   print("余额不足！")
                   return menu()
           elif ATM_money['atm']<in_money:
               print("ATM机余额不足")
               return menu()
       elif not  in_money.isdigit():
           print("请输入合法的金额！")
           return menu()

    elif choice1=='2':
       in_money=input("请输入你要存的金额：\n")
       if in_money.isdigit():
           in_money=int(in_money)
           if int(ATM_money['atm']) >= in_money:
               money=user_money['root'] +in_money
               ATM_money['atm'] = int(ATM_money['atm']) + money
               user_money['root']=money
               print("操作成功！")
               return menu()
       elif not  in_money.isdigit():
           print("请输入合法的金额！")
           return menu()
    elif choice1=='3':
        print(user_money['root'])
        return menu()
    else:return menu()



#主函数
global user_money
user_money = {"root": 5000}
global ATM_money
ATM_money={'atm':50000}
print("*"*20,"ATM机","*"*20)
print("请根据以下选项执行操作")
global user_info
user_info={}
choice=input("# 1 登入 # 2 注册\n")
if choice=='1':
    login()
elif choice=="2":
    register()
print("*"*20,"ATM机","*"*20)
